/**
 * 
 */
package co.edu.udistrital.analyzer.business.orm.vo;

import java.util.List;

import javax.faces.model.ListDataModel;

import org.primefaces.model.SelectableDataModel;

/**
 * @author Alejo
 *
 */
public class DocumentClassifyDataModel extends
		ListDataModel<DocumentClassifyDTO> implements
		SelectableDataModel<DocumentClassifyDTO> {

	/**
	 * 
	 */
	public DocumentClassifyDataModel() {
	}

	/**
	 * @param list
	 */
	public DocumentClassifyDataModel(List<DocumentClassifyDTO> list) {
		super(list);
	}

	/* (non-Javadoc)
	 * @see org.primefaces.model.SelectableDataModel#getRowData(java.lang.String)
	 */
	@Override
	public DocumentClassifyDTO getRowData(String rowKey) {
      List<DocumentClassifyDTO> documents = (List<DocumentClassifyDTO>) getWrappedData();  
        
      for(DocumentClassifyDTO document : documents) {  
          if(document.getIdDocument().equals(rowKey))  
              return document;  
      }  
        
      return null;  
	}

	/* (non-Javadoc)
	 * @see org.primefaces.model.SelectableDataModel#getRowKey(java.lang.Object)
	 */
	@Override
	public Object getRowKey(DocumentClassifyDTO document) {
		return document.getIdDocument();
	}

}
